op {
  name: "ConfigureTPUEmbedding"
  attr {
    name: "config"
    type: "string"
  }
  is_stateful: true
}
